#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2023/11/18 20:18
# @Author  : FuKai ZHANG
# @File    : excel_control.py
# @Description :excel
from common.setting import ensure_path_sep
import xlrd


# import json


def get_excel_data(sheet_name: str, case_name: any) -> list:
    """
    读取 Excel 中的数据
    :param sheet_name: excel 中的 sheet 页的名称
    :param case_name: 测试用例名称
    :return:
    这里写成了获取指定case_name的测试数据，根据需要自行修改
    """
    res_list = []

    excel_dire = ensure_path_sep("\\files\\damo.xls")
    work_book = xlrd.open_workbook(excel_dire, formatting_info=True)

    # 打开对应的子表
    work_sheet = work_book.sheet_by_name(sheet_name)
    # 读取一行
    idx = 0
    for one in work_sheet.col_values(1):  # idx跟着one一起自增
        if case_name in one:
            case_data = work_sheet.cell(idx, 3).value
            case_expected = work_sheet.cell(idx, 4).value
            res_list.append((case_name, case_data, case_expected))
        # print(one)
        idx += 1
        print(idx)
        # print(work_sheet.col_values(1))
    return res_list


if __name__ == '__main__':
    data = get_excel_data("damo", "用户名正确, 密码正确")

    print(data)
